/*

 * Copyright (c) 2014, Oracle America, Inc.

 * All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions are met:

 *

 *  * Redistributions of source code must retain the above copyright notice,

 *    this list of conditions and the following disclaimer.

 *

 *  * Redistributions in binary form must reproduce the above copyright

 *    notice, this list of conditions and the following disclaimer in the

 *    documentation and/or other materials provided with the distribution.

 *

 *  * Neither the name of Oracle nor the names of its contributors may be used

 *    to endorse or promote products derived from this software without

 *    specific prior written permission.

 *

 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"

 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE

 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR

 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF

 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS

 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN

 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)

 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF

 * THE POSSIBILITY OF SUCH DAMAGE.

 */

package org.openjdk.jmh.samples;


import org.openjdk.jmh.annotations.Benchmark;

import org.openjdk.jmh.annotations.BenchmarkMode;

import org.openjdk.jmh.annotations.Mode;

import org.openjdk.jmh.annotations.OutputTimeUnit;

import org.openjdk.jmh.annotations.Scope;

import org.openjdk.jmh.annotations.Setup;

import org.openjdk.jmh.annotations.State;

import org.openjdk.jmh.infra.BenchmarkParams;

import org.openjdk.jmh.infra.ThreadParams;

import org.openjdk.jmh.runner.Runner;

import org.openjdk.jmh.runner.RunnerException;

import org.openjdk.jmh.runner.options.Options;

import org.openjdk.jmh.runner.options.OptionsBuilder;


import java.util.ArrayList;

import java.util.List;

import java.util.concurrent.ConcurrentHashMap;

import java.util.concurrent.TimeUnit;


@BenchmarkMode(Mode.Throughput)

@OutputTimeUnit(TimeUnit.SECONDS)

@State(Scope.Benchmark)

public class JMHSample_31_InfraParams {


 /*

  * There is a way to query JMH about the current running mode. This is

  * possible with three infrastructure objects we can request to be injected:

  *   - BenchmarkParams: covers the benchmark-global configuration

  *   - IterationParams: covers the current iteration configuration

  *   - ThreadParams: covers the specifics about threading

  *

  * Suppose we want to check how the ConcurrentHashMap scales under different

  * parallelism levels. We can put concurrencyLevel in @Param, but it sometimes

  * inconvenient if, say, we want it to follow the @Threads count. Here is

  * how we can query JMH about how many threads was requested for the current run,

  * and put that into concurrencyLevel argument for CHM constructor.

  */


 static final int THREAD_SLICE = 1000;


 private ConcurrentHashMap<String, String> mapSingle;

 private ConcurrentHashMap<String, String> mapFollowThreads;


 @Setup

 public void setup(BenchmarkParams params) {

  int capacity = 16 * THREAD_SLICE * params.getThreads();

  mapSingle        = new ConcurrentHashMap<>(capacity, 0.75f, 1);

  mapFollowThreads = new ConcurrentHashMap<>(capacity, 0.75f, params.getThreads());

 }


 /*

  * Here is another neat trick. Generate the distinct set of keys for all threads:

  */


 @State(Scope.Thread)

 public static class Ids {

  private List<String> ids;


  @Setup

  public void setup(ThreadParams threads) {

   ids = new ArrayList<>();

   for (int c = 0; c < THREAD_SLICE; c++) {

    ids.add("ID" + (THREAD_SLICE * threads.getThreadIndex() + c));

   }

  }

 }


 @Benchmark

 public void measureDefault(Ids ids) {

  for (String s : ids.ids) {

   mapSingle.remove(s);

   mapSingle.put(s, s);

  }

 }


 @Benchmark

 public void measureFollowThreads(Ids ids) {

  for (String s : ids.ids) {

   mapFollowThreads.remove(s);

   mapFollowThreads.put(s, s);

  }

 }


 /*

  * ============================== HOW TO RUN THIS TEST: ====================================

  *

  * You can run this test:

  *

  * a) Via the command line:

  *    $ mvn clean install

  *    $ java -jar target/benchmarks.jar JMHSample_31 -t 4 -f 5

  *    (we requested 4 threads, and 5 forks; there are also other options, see -h)

  *

  * b) Via the Java API:

  *    (see the JMH homepage for possible caveats when running from IDE:

  *      http://openjdk.java.net/projects/code-tools/jmh/)

  */


 public static void main(String[] args) throws RunnerException {

  Options opt = new OptionsBuilder()

          .include(JMHSample_31_InfraParams.class.getSimpleName())

          .threads(4)

          .forks(5)

          .build();


  new Runner(opt).run();

 }

// 本例介绍了在方法中可覆盖的三种参数，这给在测试时获取配置以及动态修改配置提供了可能
//
// BenchmarkParams：基准测试级别IterationParams：迭代级别ThreadParams：线程级别
}